package com.xizi.sort;

import java.util.Arrays;

public class CountSort {

    public static void main(String[] args) {
        int[] arr = new int[]{2, 5, 3, 0, 2, 3, 0, 3};
        System.out.println("排序前---------------------");
        System.out.println(Arrays.toString(arr));
        int[] res = countSort01(arr);
        System.out.println("排序后---------------------");
        System.out.println(Arrays.toString(res));
    }


    public static int[] countSort(int[] arr) {
        // 获取待排序数列的最大值,与最小值 用来创建辅助数组
        int min = arr[0], max = arr[0];
        for (int i = 0; i < arr.length; i++) {
            max = Math.max(max, arr[i]);
            min = Math.min(min, arr[i]);
        }
        // 创建辅助数组 先在辅助数组存储其对应下标在待排序数中出现的次数 (与countSort01同)
        int[] sortArr = new int[max + 1];
        sortArr = new int[max - min + 1];
        for (int i = 0; i < arr.length; i++) {
            sortArr[arr[i] - min]++;
        }
        // 对辅助数组进行改进 将其存储的对应待排序数列的次数 转换为临时排序数列的下标位置
        for (int i = 1; i < sortArr.length; i++) {
            sortArr[i] += sortArr[i - 1];
        }
        // 创建临时排序数列
        int[] tsortArr = new int[arr.length];
        for (int i = arr.length - 1; i > 0; i--) {
            tsortArr[--sortArr[arr[i] - min]] = arr[i];
        }
        return tsortArr;
    }

    public static int[] countSort02(int[] arr) {
        // 获取待排序数列的最大值 用来创建辅助数组
        int max = arr[0];
        for (int i = 0; i < arr.length; i++) {
            max = Math.max(max, arr[i]);
        }
        // 创建辅助数组 先在辅助数组存储其对应下标在待排序数中出现的次数 (与countSort01同)
        int[] sortArr = new int[max + 1];
        for (int i = 0; i < arr.length; i++) {
            sortArr[arr[i]]++;
        }
        // 对辅助数组进行改进 将其存储的对应待排序数列的次数 转换为临时排序数列的下标位置
        for (int i = 1; i < sortArr.length; i++) {
            sortArr[i] += sortArr[i - 1];
        }
        // 创建临时排序数列 并从右往左遍历arr将其顺序放于tsortArr
        int[] tsortArr = new int[arr.length];
        for (int i = arr.length - 1; i > 0; i--) {
            tsortArr[--sortArr[arr[i]]] = arr[i];
        }
        return tsortArr;
    }

    public static int[] countSort01(int[] arr) {
        // 获取序列的最大值
        int max = arr[0];
        int min = arr[0];
        for (int i = 0; i < arr.length; i++) {
            max = Math.max(max, arr[i]);
            min = Math.min(min, arr[i]);
        }
        // 创建一个长度为max+1的辅助数组 用来存储与辅助数组对应下标的数值 在待排序数列中出现的次数
        int[] sortArr = new int[max - min + 1];
        for (int i = 0; i < arr.length; i++) {
            // 遍历带排序数列 每次将辅助序列对应下标处的数值+1
            sortArr[arr[i]]++;
        }
        int[] res = new int[arr.length];
        int index = 0;
        for (int i = 0; i < sortArr.length; i++) {
            for (int j = 0; j < sortArr[i]; j++) {
                res[index++] = i;
            }
        }
        return res;
    }

}